#!/bin/bash
###
 # @Version: 2.0
 # @Author: Yue Zhong
 # @Date: 2024-09-26 20:02:26
 # @Description: 
 # @LastEditors: Yue Zhong
 # @LastEditTime: 2024-11-11 16:59:21
### 

gpu=0
dataset_dir='./data'
dataset_name=CIFAR10 #Tiny-ImageNet-200
batch_size=96
T=8
channels=64
suffix='_SNN_Adam_600ep_2022/checkpoint.pth.tar'
search_space=AutoSNN_$channels
prefix='macro_search_result2/uniform_sampling/'$search_space'_'
trained_supernet=$prefix$dataset_name$suffix

python train_supernet/train.py \
    --gpu $gpu \
    --T $T --init_tau 2.0 --v_threshold 1.0 --neuron PLIF \
    --epochs 600 \
    --dataset_dir $dataset_dir \
    --dataset_name $dataset_name \
    --save uniform_sampling \
    --search_space $search_space \
    --seed 2022 \
    --batch_size $batch_size
echo "Training supernet on $dataset_name is completed"
